package com.gitee.deament.server.core.cli.db.hash;

import com.gitee.deament.server.config.SystemConfig;
import com.gitee.deament.server.core.annotation.CliHander;
import com.gitee.deament.server.core.cli.AbstractCli;
import com.gitee.deament.server.core.data.ConnectData;
import com.gitee.deament.server.core.data.datatype.Value;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;

import java.util.List;
import java.util.Map;

/**
 * HLEN
 * HLEN key
 *
 * 返回哈希表 key 中域的数量。
 *
 * 时间复杂度：
 * O(1)
 *
 * 返回值：
 * 哈希表中域的数量。
 * 当 key 不存在时，返回 0 。
 * @author deament
 */
@CliHander(command = "HLEN")
public class HLENcli extends AbstractCli {
    @Override
    public void executeCli(ChannelHandlerContext channelHandlerContext, String command) {
        List<String> values = getValues(command);
        if (values.size() != 3) {
            channelHandlerContext.channel().writeAndFlush(new TextWebSocketFrame("ERR wrong number of arguments for 'HLEN' command"));
            return;
        }
        String hash = values.get(1);
        int db = ConnectData.SELECT_DB_STORAGE.getUseDB(channelHandlerContext.channel().id().asLongText());
        Value<Map<String, String>> hashData = SystemConfig.dbData.getHashData(db, hash);
        if(hashData==null || hashData.getValue()==null){
            channelHandlerContext.channel().writeAndFlush("0");
        }
        channelHandlerContext.channel().writeAndFlush(new TextWebSocketFrame(hashData.getValue().size()+""));
    }
}
